백준 9933 민균이의 비밀번호

백준 9933 민균이의 비밀번호는 주어지는 문자열이 주어진 문자열 집합에서 거꾸로 된 것과 같은 것이 있는지 확인하고 찾아 출력하는 문제이다. 너무 간단한 문제라고 생각하고 주어진 문자열에서 뒤집어서 같은 문자열이 있는지만 확인했는데 오답이 나와 무엇이 잘못됬는지 잘 몰랐다.

이 문제에는 예외가 존재하는데 길이가 1일 때와 여러개의 문자열이 있어도 문자열 1개 자체가 펠린드롬(앞으로 읽어도 거꾸로 읽어도 같은 문제)일 경우가 존재하면 그 문자가 답이 되어 버린다.

아래의 소스코드는 문자열이 펠린드롬인지 확인하고 펠린드롬이 아니면 문자열의 원래 형태와 거꾸로 된 형태를 set에 넣는다. 그리고 set에서 현재 문자열과 거꾸로 뒤집은 문자열의 set이 있는지 확인하고 있으면 답으로 출력하는 코드이다.

#pragma warning(disable:4996)
#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<queue>
#include<algorithm>
#include<set>
#define INF 987654321
using namespace std;

int n;
set<string> s;
string a;

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	//freopen("input.txt", "r", stdin);
	cin >> n;

	string result;
	for (int i = 0; i < n; i++)
	{
		cin >> a;
		string b(a);
		reverse(b.begin(), b.end());

		if (a.compare(b) == 0)
		{
			result = a;
			break;
		}

		if (s.find(a) != s.end())
		{
			result = a;
			break;
		}
		if (s.find(b) != s.end())
		{
			result = b;
			break;
		}

		s.insert(a);
		s.insert(b);
	}
	
	cout << result.length() << ' ' << result[result.length() / 2];
}